python assignment 2 room area